package com.jfinal.bbsapp.common.model;

import com.jfinal.bbsapp.common.config.CommonConfig;
import com.jfinal.plugin.activerecord.dialect.MysqlDialect;
import com.jfinal.plugin.activerecord.generator.Generator;
import com.jfinal.plugin.druid.DruidPlugin;

import javax.sql.DataSource;

/**
 * Model、BaseModel、_MappingKit 生成器
 */
public class _Generator {
	/**
	 * 在此统一添加不参与生成的 table
	 */
	private static final String[] blacklist = {
		"sys_log_202506","sys_log_202507", "sys_log_202508", "sys_log_202509", "sys_log_202510","sys_log_202511", "sys_log_202512",
		"c_user_record_202506","c_user_record_202507", "c_user_record_202508", "c_user_record_202509", "c_user_record_202510","c_user_record_202511", "c_user_record_202512"
	};

	/**
	 * 重用 JFinalClubConfig 中的数据源配置，避免冗余配置
	 */
	public static DataSource getDataSource() {
		DruidPlugin druidPlugin = CommonConfig.getDruidPlugin();
		druidPlugin.start();
		return druidPlugin.getDataSource();
	}

	public static void main(String[] args) {
		// base model 所使用的包名
		String baseModelPackageName = "com.jfinal.bbsapp.common.model.base";
		// base model 文件保存路径
		String baseModelOutputDir = System.getProperty("user.dir") + "/src/main/java/com/jfinal/bbsapp/common/model/base";
		System.out.println("输出路径：" + baseModelOutputDir);

		// model 所使用的包名 (MappingKit 默认使用的包名)
		String modelPackageName = "com.jfinal.bbsapp.common.model";
		// model 文件保存路径 (MappingKit 与 DataDictionary 文件默认保存路径)
		String modelOutputDir = baseModelOutputDir + "/..";

		// 创建生成器
		Generator gen = new Generator(getDataSource(), baseModelPackageName, baseModelOutputDir, modelPackageName, modelOutputDir);

		// 设置数据库方言
		gen.setDialect(new MysqlDialect());

		// 设置是否生成字段备注
		gen.setGenerateRemarks(true);

		// 添加不需要生成的表名到黑名单
		gen.addBlacklist(blacklist);

		// 设置是否在 Model 中生成 dao 对象
		gen.setGenerateDaoInModel(false);

		// 设置是否生成字典文件
		gen.setGenerateDataDictionary(false);

		// 设置需要被移除的表名前缀用于生成modelName。例如表名 "osc_user"，移除前缀 "osc_"后生成的model名为 "User"而非
		// OscUser
		// gernerator.setRemovedTableNamePrefixes("t_");

		// 生成
		gen.generate();
	}
}
